package cn.dmp.tags

import org.apache.commons.lang.StringUtils
import org.apache.spark.sql.Row

object Tags4Ad  extends Tags {
    /**
      * 行数据 -》 标签数据
      *
      * @param args
      * @return
      */
    override def makeTags(args: Any*): Map[String, Int] = {

        val row = args(0).asInstanceOf[Row]

        // 存放当前行的标签数据
        var tags = Map[String, Int]()

        val adST = row.getAs[Int]("adspacetype")
        val adSTN = row.getAs[String]("adspacetypename")

        // 广告位类型 & 广告位名称
        if (adST > 10) tags += "LC"+adST -> 1 else tags += "LC0"+adST -> 1
        if (StringUtils.isNotEmpty(adSTN)) tags += "LN"+adSTN -> 1



        // 渠道
        val channel = row.getAs[Int]("adplatformproviderid")
        if (channel > 0) tags += "CN"+channel -> 1


        tags
    }
}
